from pages.LoginPage import LoginPage
from common.GetLoc import SystemManageLoc


class SystemManage(LoginPage):
    loc_cityList = SystemManageLoc().get_loc_cityList()
    loc_companyList = SystemManageLoc().get_loc_companyList()
    loc_departmentList = SystemManageLoc().get_loc_departmentList()
    loc_productList = SystemManageLoc().get_loc_productList()
    loc_channelList = SystemManageLoc().get_loc_channelList()
    loc_sourceList = SystemManageLoc().get_loc_sourceList()
    loc_lastContent = SystemManageLoc().get_loc_lastContent()
    loc_checkBox = SystemManageLoc().get_loc_checkBox()
    loc_addButton = SystemManageLoc().get_loc_addButton()
    loc_editButton = SystemManageLoc().get_loc_editButton()
    loc_delButton = SystemManageLoc().get_loc_delButton()
    loc_confirmButton = SystemManageLoc().get_loc_confirmButton()
    loc_alertButton = SystemManageLoc().get_loc_alertButton()
    loc_infoList = SystemManageLoc().get_loc_infoList()

    def find_info(self,text):
        way = self.loc_infoList[0]
        loc = self.loc_infoList[1] % text
        return self.find_ele((way, loc)).text

    def chose_checkBox(self,text):
        way = self.loc_checkBox[0]
        loc = self.loc_checkBox[1] % text
        self.find_ele((way, loc)).click()
